package com.leetcode;

/**
 * 482. 密钥格式化
 *
 * @author yml
 * 有一个密钥字符串 S ，只包含字母，数字以及 '-'（破折号）。其中， N 个 '-' 将字符串分成了 N+1 组。
 * <p>
 * 给你一个数字 K，请你重新格式化字符串，使每个分组恰好包含 K 个字符。特别地，第一个分组包含的字符个数必须小于等于 K，但至少要包含 1 个字符。两个分组之间需要用 '-'（破折号）隔开，并且将所有的小写字母转换为大写字母。
 * <p>
 * 给定非空字符串 S 和数字 K，按照上面描述的规则进行格式化。
 */
public class LeetCode482 {

    public static void main(String[] arg0) {
        System.out.println(licenseKeyFormatting("2-5g-3-J", 2));
    }

    public static String licenseKeyFormatting(String s, int k) {
        String upperStr = s.toUpperCase().replace("-", "");
        int length = upperStr.length();
        if (length < k) {
            return upperStr;
        }

        int remainder = length % k;
        StringBuilder resSB = new StringBuilder();
        if (remainder != 0) {
            resSB.append(upperStr.subSequence(0, remainder));
            resSB.append("-");
        }

        for (int i = remainder; i < length; i += k) {
            if (i != remainder) {
                resSB.append("-");
            }
            resSB.append(upperStr.subSequence(i, i + k));
        }

        return resSB.toString();
    }

}
